package frame

import (
	"stormragetech/dawn/data"

	"gopkg.in/mgo.v2/bson"
)

func init() {
	data.GetConn("test", "sys_corp")
}

// Corp 公司 结构体
type Corp struct {
	ID     bson.ObjectId `bson:"_id"`
	CorpID string        `bson:"corp_id"`
	Name   string        `bson:"name"`
	Locked string        `bson:"locked"`
}

// InsCorp 新增公司
func InsCorp(c Corp) error {
	return data.Conn.Insert(c)
}

// FindByCorpID 通过ID查询公司
func FindByCorpID(id string) (Corp, error) {
	c := Corp{}
	err := data.Conn.Find(bson.M{"corp_id": id}).One(&c)
	return c, err
}
